import os
import uuid
from datetime import datetime

from data_analysis_agent import LLMConfig
from data_analysis_agent.utils import LLMHelper


def create_session_output_dir(base_output_dir,user_input: str,analysys_files: list) -> str:
    """为本次分析创建独立的输出目录"""

    # 使用UUID创建唯一的会话目录名（16进制格式，去掉连字符）
    session_id = uuid.uuid4().hex

    llm = LLMHelper(LLMConfig())
    prompt = "这几张表是关于哪一个公司的？请直接返回公司名称即可。" + "\n".join(analysys_files)
    company_name = llm.call(prompt,temperature=0.05)

    dir_name = f"session_{company_name}_{session_id}"
    print(f"保存结果目录：{dir_name}")
    session_dir = os.path.join(base_output_dir, dir_name)
    os.makedirs(session_dir, exist_ok=True)
    return session_dir



